/* Copyright (c) 2021, National University of Defense Technology. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef PAR_COMPOUND_OPERATOR_H
#define PAR_COMPOUND_OPERATOR_H

#include "par_operator.h"

namespace YHAMG
{

class Par_Compound_Operator : public Par_Operator
{
private:
	const Par_Operator* A;
	const Par_Operator* B;
	const Par_Operator* C;
	double alpha;
	double beta;
	double gamma;

public:	
	Par_Compound_Operator(double alpha, const Par_Operator& A, double beta);	
	Par_Compound_Operator(double alpha, const Par_Operator& A, double beta, const Par_Operator& B, double gamma);
	Par_Compound_Operator(double alpha, const Par_Operator& A, const Par_Operator& B, double gamma);
	Par_Compound_Operator(double alpha, const Par_Operator& A, const Par_Operator& B, double beta, const Par_Operator& C, double gamma);
	int InSize() const;
	int OutSize() const;
	void Apply(const Par_Vector& x, const Par_Vector& y) const;
	void Apply(const Par_MultiVector& X, const Par_MultiVector& Y) const;
};

}

#endif